package com.shj.util;

import org.apache.commons.io.FileUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;

/**
 * @author songhuajian
 * @Description: TODO 文件工具类
 * @date 2020/6/5 15:14
 **/
public class FileUtil {
    // 保存目录
    public static final String SAVE_URL = "G:\\";
    // 保存的名字

    /**
     * @Description 保存文件返回路径
     * @author songhuajian
     * @date 2020/6/6 14:08
     */
    public static String fileUpload(File file, String fileFileName){
        File savefile = null;
        try {
            //文件名（时间+文件名字）
            savefile = new File(new File(SAVE_URL), MyTimeUtil.getForString()+fileFileName);
            //判断上传文件的保存目录是否存在
            if(!savefile.getParentFile().exists()){
                savefile.getParentFile().mkdirs();
            }
            FileUtils.copyFile(file, savefile);
        } catch (IOException e) {
            e.printStackTrace();
        }finally{
            if(file.exists()){
                file.delete();
            }
        }
        //返回保存文件路径
        return savefile.getAbsolutePath();
    }

    /**
     * @Description mvc保存文件，返回保存路径
     * @author songhuajian
     * @date 2020/6/7 15:28
     */
    public static String mvcFileUpload(MultipartFile file) throws IOException {
        // 文件名
        String fileName = file.getOriginalFilename();
        assert fileName != null;
        String suffixName = fileName.substring(fileName.lastIndexOf("."));
        // 保存路径
        String filePath = FileUtil.SAVE_URL;
        // 新文件名(时间+随机五位数字)
        fileName = MyIdUtil.getForTimeId() + suffixName;
        File dest = new File(filePath + fileName);
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        // 保存文件
        file.transferTo(dest);
        // 保存路径转义
        return StringUtil.deal(filePath + fileName);
    }
}
